import torch

# 查询GPU数量
gpu_count = torch.cuda.device_count()
print(f"GPU数量: {gpu_count}")

# 列出每个GPU的名称
for i in range(gpu_count):
    print(f"GPU {i} 名称: {torch.cuda.get_device_name(i)}")
